tensor([[[0.3900, 0.3900],
         [0.3900, 0.3900]],

        [[0.3200, 0.3200],
         [0.3200, 0.3200]],

        [[0.2900, 0.2900],
         [0.2900, 0.2900]]], grad_fn=<RoundBackward1>)